py_library(
    name = "fla",
    srcs = glob(["fla/*.py"]),
    visibility = ["//visibility:public"]
)

py_library(
    name = "causal_conv1d",
    srcs = glob(["causal_conv1d/*.py"]),
    visibility = ["//visibility:public"]
)

py_library(
    name = "moe",
    srcs = glob(["moe/*.py"]),
    visibility = ["//visibility:public"]
)

py_library(
    name = "common",
    srcs = glob(["common/*.py"]),
    visibility = ["//visibility:public"]
)

py_library(
    name = "triton_kernels",
    deps = [
        ":fla",
        ":causal_conv1d",
        ":moe",
        ":common",
    ],
    visibility = ["//visibility:public"]
)